/*******************************************************************

    Author: Xiao
    Date: 2022/3/28 10:44

*******************************************************************/
package dao

import (
	"context"
	"fmt"
	"github.com/go-redis/redis/v8"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"gorm.io/gorm/schema"
	"log"
	"mcr/pkg/config"
	"sync"
	"time"
)

var (
	MDB *gorm.DB
	Rdb *RedisDb
)

type RedisDb struct {
	rdb *redis.Client
	ctx context.Context
	mut sync.RWMutex
}

// 初始化数据库连接
func InitFactory(){
	if err := ConnectMDb();err != nil{
		log.Fatalln(err)
	}

	if err := connectRDB();err != nil{
		log.Fatalln(err)
	}
}

// 连接mysl
func ConnectMDb() error{
	dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
		config.ConfigValue.MDBConfig.Username, config.ConfigValue.MDBConfig.Password, config.ConfigValue.MDBConfig.Host, config.ConfigValue.MDBConfig.Port, config.ConfigValue.MDBConfig.Name)
	var err error
	MDB,err = gorm.Open(mysql.Open(dsn),&gorm.Config{
		//	去除复数表名,和表名前缀
		NamingStrategy: schema.NamingStrategy{
				SingularTable: true,
				TablePrefix: "sa_",
			},

	})
	if err != nil{
		return fmt.Errorf("连接Mysql数据库失败:%s",err)
	}

	// 获取通用数据库对象 sql.DB ，然后使用其提供的功能
	sqlDB, err := MDB.DB()
	if err != nil{
		return fmt.Errorf("获取DB失败:%s",err)
	}
	// SetMaxIdleConns 用于设置连接池中空闲连接的最大数量。
	sqlDB.SetMaxIdleConns(10)

	// SetMaxOpenConns 设置打开数据库连接的最大数量。
	sqlDB.SetMaxOpenConns(100)

	// SetConnMaxLifetime 设置了连接可复用的最大时间。
	sqlDB.SetConnMaxLifetime(time.Hour)
	return nil
}

// 连接redis
func connectRDB() error {
	Rdb = new(RedisDb)
	Rdb.ctx = context.Background()
	rdb := redis.NewClient(&redis.Options{
		Addr:     fmt.Sprintf("%s:%s",config.ConfigValue.RDBConfig.Host,config.ConfigValue.RDBConfig.Port),
		Password: config.ConfigValue.RDBConfig.Password, // no password set
		DB:       config.ConfigValue.RDBConfig.Select,  // use default DB
	})
	Rdb.rdb = rdb
	_,err := Rdb.rdb.Ping(Rdb.ctx).Result()
	if err != nil{
		return fmt.Errorf("连接Redis失败:%s",err)
	}
	return nil
}